package com.hyyt.shopjsp.util.tree;

import com.hyyt.shopjsp.util.common.LoggerUtils;

import java.util.*;

/**
 * @author LQS on 2017/8/11.
 * NodeUitl:封装bootstrap树形节点帮助类
 */
public class NodeUtil {

    /**
     * 获取节点的json字符串
     * @param nodeList 节点列表数据
     * @return
     */
    public static String  getNodeJson(List  nodeList,String rootName){
        // 节点列表（散列表，用于临时存储节点对象）
        Map nodeMap = new LinkedHashMap();
        // 根节点
        Node root = new  Node();
        root.setId("0");
        root.setText(rootName);
        root.setParentId("");
        root.setIsLeaf("0");
        nodeMap.put(root.getId(), root);
        // 根据结果集构造节点列表（存入散列表）
        for (Iterator it = nodeList.iterator(); it.hasNext();) {
            Map dataRecord = (Map) it.next();
            Node node = new Node();
            node.setId(dataRecord.get("id").toString());
            node.setText(dataRecord.get("text").toString());
            node.setIsLeaf(dataRecord.get("isLeaf").toString());
            node.setParentId(dataRecord.get("parentId").toString());
            nodeMap.put(node.getId(), node);
        }
        // 构造无序的多叉树
        Set entrySet = nodeMap.entrySet();
        for (Iterator it = entrySet.iterator(); it.hasNext();) {
            Node node = (Node) ((Map.Entry) it.next()).getValue();
            if (node.getParentId() == null || node.getParentId().equals("")) {
                root = node;
            } else {
                String parentId=node.getParentId();
                try{
                    Node nodeTemp=(Node)nodeMap.get(parentId);
                    nodeTemp.addChild(node);
                }catch (Exception e){
                    LoggerUtils.error(NodeUtil.class,e+"，多叉树异常数据："+node.toString());
                }

            }
        }
        // 输出无序的树形菜单的JSON字符串
        return "["+root.toString()+"]";
    }


}
